package _01_sort;

import org.junit.Test;

import java.util.Arrays;
import java.util.concurrent.ThreadLocalRandom;

import static _01_sort.SortUtil.swap;

/**
 * @author: mornd
 * @dateTime: 2023/6/23 - 5:50
 * 双边循环快排（霍尔分区）
 * 1、选择最左侧元素作为基准点元素
 * 2、j指针负责从右向左找比基准点小或等的元素，i指针负责从左向右找比基准点大的元素，一旦找到二者交换元素，直至i，j相交
 * 3、最后基准点与i（此时i与j相等）交换，i即为分区位置
 */
public class QuickSortHoare {
    @Test
    public void test() {
        int[] arr = {2, 7, 1, 9, 3, 4, 11, 46};
//        int[] arr = {5, 4, 3, 2, 1};

        quick(arr, 0, arr.length - 1);
        System.out.println(Arrays.toString(arr));
    }

    public void quick(int[] arr, int left, int right) {
        if (left >= right) {
            return;
        }
        int pv = partition(arr, left, right);
        quick(arr, left, pv - 1);
        quick(arr, pv + 1, right);
    }

    // 分区
    private int partition(int[] arr, int left, int right) {
        // 使用随机数作做基准点
//        int randomIdx = left + ThreadLocalRandom.current().nextInt(right - left + 1);
//        swap(arr, randomIdx, left);
        int pv = arr[left];
        int i = left;
        int j = right;
        while (i < j) {
            while (arr[j] > pv) {
                // 从右向左找比基准点小(等于)的元素
                j--;
            }
            while (i < j && arr[i] <= pv) {
                // 从左向右找比基准点大的元素
                i++;
            }
            if (i < j) { // 这里和最后的if判断只是为了去掉多余的交换，去掉不影响最终结果
                swap(arr, i, j);
            }
        }
        // 最后的i下标，就是基准点的位置，将i与原始基准点交换
        if (i != left) {
            swap(arr, i, left);
        }
        return i;
    }
}
